#!/usr/bin/env python
# -*- coding: utf-8 -*-

import requests

# 方法一: 使用首页的响应对象中的cookies属性获取cookie, 并在后面的请求中带上它, 方可让状态连续(后面的请求有登陆成功的状态)
# 首页
# url = 'http://localhost:8888/smeoa/login'
# r = requests.get(url)   #响应对象
# cookies  = r.cookies    #获取响应cookie
#
# #登陆
# url = 'http://localhost:8888/smeoa/login/check_login'   #构造URL
# params = {'emp_no':'admin', 'password':'admin'} #构造正文, 字典格式
# r = requests.post(url=url, data=params, cookies = cookies)     #发送post请求
#
# #断言
# if '退出' in r.text:
#     print('登陆成功')
# else:
#     print('登陆失败')
#
# # 日报首页
# url = 'http://localhost:8888/smeoa/task/index'
# r = requests.post(url, cookies = cookies)
# print(r.text)

# 方法二:
s = requests.session()  # 使用Session的类生成一个会话对象, 代码会自动维护cookie
url = 'http://localhost:8888/smeoa/login'
r = s.get(url)  # 发送请求时使用s这个会话对话, 自动维护cookie

url = 'http://localhost:8888/smeoa/login/check_login'
params = {'emp_no': 'admin', 'password': 'admin'}
r = s.post(url, data=params)

if '退出' in r.text:
    print('登陆成功')
else:
    print('登陆失败')

url = 'http://localhost:8888/smeoa/task/index'
r = s.post(url)
if '所有任务' in r.text:
    print('获取日报首页成功')
else:
    print('获取日报首页失败')
